package com.jielin.handling.util.videoutil;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import com.jielin.handling.util.ConstantUtil;
import com.jielin.handling.util.videoutil.Utils;

public class FFMPEG {

	public static HashMap<String, String> builder() {// 初始化参数
		HashMap<String, String> dto = new HashMap<String, String>();
		if (Utils.getOsPrefix().toLowerCase().equals("linux-amd64")) {
			dto.put("ffmpeg_path", ConstantUtil.FFMPEG_PATH_LINUX_AMD64);// 必填
		} else if (Utils.getOsPrefix().toLowerCase().equals("linux-i386")) {
			dto.put("ffmpeg_path", ConstantUtil.FFMPEG_PATH_LINUX_I386);// 必填
		} else if (Utils.getOsPrefix().toLowerCase().equals("win32-amd64")) {
			dto.put("ffmpeg_path", ConstantUtil.FFMPEG_PATH_WIN32_AMD64);// 必填
		} else if (Utils.getOsPrefix().toLowerCase().equals("win32-x86")) {
			dto.put("ffmpeg_path", ConstantUtil.FFMPEG_PATH_WIN32_X86);// 必填
		}
//		dto.put("input_path", "/data/dossier/RecordVideoFile/2019_6_26_72082423.mp4");// 必填
//		dto.put("video_converted_path", "/data/dossier/RecordVideoFile/out/2019_6_26_72082423.mp4");// 必填
		dto.put("logo", ConstantUtil.WATERMARKIMAGEG_PATH);// 可选(注意windows下面的logo地址前面要写4个反斜杠,如果用浏览器里面调用servlet并传参只用两个,如
		dto.put("xaxis", "main_w-overlay_w");// 右上角为 overlay= main_w-overlay_w:0
		dto.put("yaxis", "0");
		return dto;
	}

	public static String dealString(String str) {
		Matcher m = java.util.regex.Pattern.compile("^frame=.*").matcher(str);
		String msg = "";
		while (m.find()) {
			msg = m.group();
		}
		return msg;
	}

	/**
	 * 如果是数字就是成功的时间(秒数)
	 * 
	 * @param str
	 * @return
	 */
	public static boolean isNumeric(String str) {
		Pattern pattern = Pattern.compile("[0-9]*");
		Matcher isNum = pattern.matcher(str);
		if (!isNum.matches()) {
			return false;
		}
		return true;
	}

	/**
	 * 如果返回不是null的值就是成功(值为转换用时单位:秒)
	 * 
	 * @param instr
	 * @return
	 */
	public static String returnSecond(String instr) {
		String returnValue = null;
		if (null != instr) {
			String[] a = instr.split("\\.");
			String[] b = a[0].split(":");
			int returnNumber = 0;
			if (null != instr && b[0].length() != 0) {
				returnNumber = Integer.valueOf(b[0]) * 60 * 60 + Integer.valueOf(b[1]) * 60 + Integer.valueOf(b[2]);
				returnValue = String.valueOf(returnNumber);
			} else {
				returnValue = null;
			}
		}
		return returnValue;
	}

	/**
	 * 获取视频格式(转码前的格式和转码后的格式都可以调用)
	 * 
	 * @param outputPath
	 * @return
	 */
	public static String returnVideoFormat(String outputPath) {
		return outputPath.substring(outputPath.lastIndexOf(".") + 1);
	}

	/**
	 * @ HashMap<String,String> dto 参数传递对象<br>
	 * dto中包含的参数<br>
	 * (必填)1.ffmpeg_path:ffmpeg执行文件地址,如 d:\\ffmpeg\\ffmpeg.exe
	 * Linux下直接调用ffmpeg命令(当然你事先已经有这个程序了)<br>
	 * (必填)2.input_path:原视频路径<br>
	 * (必填)3.video_converted_path:转换后视频输出路径<br>
	 * (可选)4.screen_size:视频尺寸 长度乘宽度 乘号用英文小写"x"如 512x480<br>
	 * (可选)5.logo:水印地址(其实在ffmpeg中有一个专门的watermark参数,logo跟它有何不同,我还没看,不过对我来说效果一样
	 * 貌似需要png图片才行)<br>
	 * (可选,如果填写必须有logo才行,默认为0)6.xaxis:水印logo的横坐标(只有logo参数为一个正确路径才行) 比如0<br>
	 * (可选,如果填写必须有logo才行,默认为0)6.yaxis:水印logo的纵坐标(只有logo参数为一个正确路径才行) 比如0<br>
	 * (可选)vb:视频比特率,传入一个数值,单位在程序里面拼接了k (可选)ab:音频比特率,传入一个数值,单位在程序里面拼接了k
	 */
	public static String videoTransfer(HashMap<String, String> dto) {
//		   String ffmpeg_path,String input_path,String video_converted_path,String logo,String screen_size,String xaxis,String yaxis,String vb,String ab
		List<String> cmd = new ArrayList<String>();
		cmd.add(dto.get("ffmpeg_path"));
		cmd.add("-y");
		cmd.add("-i");
		cmd.add(dto.get("input_path"));
		if (null != dto.get("screen_size")) {
			cmd.add("-s");
			cmd.add(dto.get("screen_size"));
		}
		if (null != dto.get("logo")) {
			String logo = dto.get("logo");
			cmd.add("-vf");
			String xaxis = dto.get("xaxis");
			String yaxis = dto.get("yaxis");
			xaxis = xaxis != null && !xaxis.equals("") ? xaxis : "0";
			yaxis = yaxis != null && !yaxis.equals("") ? yaxis : "0";

			String logoString = "movie=" + logo + "[logo],[in][logo]overlay=x=" + xaxis + ":y=" + yaxis + "[out]";
			cmd.add(logoString);
		}
		cmd.add("-strict");
		cmd.add("-2");
		if (null != dto.get("vb") && !dto.get("vb").equals("")) {
			cmd.add("-vb");
			cmd.add(dto.get("vb") + "k");
		}
		if (null != dto.get("ab") && !dto.get("ab").equals("")) {
			cmd.add("-ab");
			cmd.add(dto.get("ab") + "k");
		}
		cmd.add("-q:v");
		cmd.add("4");
		cmd.add(dto.get("video_converted_path"));
		String converted_time = CmdExecuter.exec(cmd);
		return returnSecond(converted_time);// 获取转换时间
	}

	public static void main(String[] arg) {
		HashMap<String, String> dto = new HashMap<String, String>();
//		dto.put("ffmpeg_path", "E:\\迅雷下载\\ffmpeg\\ffmpeg.exe");// 必填
//		dto.put("input_path", "E:\\迅雷下载\\ffmpeg\\input\\gg.mp4");// 必填
//		dto.put("video_converted_path", "E:\\迅雷下载\\ffmpeg\\output\\gg.mp4");// 必填
//		dto.put("logo", "E\\\\:/迅雷下载/ffmpeg/input/logo.png");// 可选(注意windows下面的logo地址前面要写4个反斜杠,如果用浏览器里面调用servlet并传参只用两个,如
//																// d:\\:/ffmpeg/input/logo.png)

//		dto.put("ffmpeg_path", "C:\\Users\\WIN10\\Downloads\\ffmpeg-20190701-e51cc7e-win64-static\\bin\\ffmpeg.exe");// 必填
//		dto.put("input_path", "C:\\data\\dossier\\RecordVideoFile\\2019_6_26_72082423.mp4");// 必填
//		dto.put("video_converted_path", "C:\\data\\dossier\\RecordVideoFile\\out\\2019_6_26_72082423.mp4");// 必填
//		dto.put("logo", "C\\\\:/data/logo.png");// 可选(注意windows下面的logo地址前面要写4个反斜杠,如果用浏览器里面调用servlet并传参只用两个,如
//		dto.put("xaxis", "main_w-overlay_w");// 　右上角为 overlay= main_w-overlay_w:0
//		dto.put("yaxis", "0");

		System.out.println(Utils.getOsPrefix().toLowerCase());

		if (Utils.getOsPrefix().toLowerCase().equals("linux-amd64")
				|| Utils.getOsPrefix().toLowerCase().equals("win32-amd64")) {
			dto.put("ffmpeg_path", "/data/ffmpeg-Linux/ffmpeg-git-20190629-amd64-static/ffmpeg");// 必填
		} else if (Utils.getOsPrefix().toLowerCase().equals("linux-i386")
				|| Utils.getOsPrefix().toLowerCase().equals("win32-x86")) {
			dto.put("ffmpeg_path", "/data/ffmpeg-Linux/ffmpeg-git-20190629-i686-static/ffmpeg");// 必填
		}
		dto.put("input_path", "/data/dossier/RecordVideoFile/2019_6_26_72082423.mp4");// 必填
		dto.put("video_converted_path", "/data/dossier/RecordVideoFile/out/2019_6_26_72082423.mp4");// 必填
		dto.put("logo", "/data/logo.png");// 可选(注意windows下面的logo地址前面要写4个反斜杠,如果用浏览器里面调用servlet并传参只用两个,如
		dto.put("xaxis", "main_w-overlay_w");// 右上角为 overlay= main_w-overlay_w:0
		dto.put("yaxis", "0");

		String secondsString = new FFMPEG().videoTransfer(dto);
		System.out.println("转换共用:" + secondsString + "秒");
	}
}